import argparse
from predict import predict_model


def main(args=None):
    parser = argparse.ArgumentParser()
    default_str = 'default: %(default)s'
    parser.add_argument('name', type=str)
    parser.add_argument('--candidate_file', required=True)
    parser.add_argument('--candidate_vol_file', required=True)
    parser.add_argument('--modelname', default='vgg13', help=default_str)
    parser.add_argument('--batchsize', type=int, default=32, help=default_str)
    parser.add_argument('--crop_shape', type=int, nargs=3, default=[64, 64, 64], help=default_str)
    parser.add_argument('--base_nb_filters', type=int, default=32, help=default_str)
    parser.add_argument('--weight_decay', type=float, default=1e-4, help=default_str)
    parser.add_argument('--loadfrom', required=True)
    namespace = parser.parse_args(args)
    predict_model(**vars(namespace))
